from aiogram.fsm.state import State, StatesGroup


class AdminCmd(StatesGroup):
    main = State()
    work_shift = State()


class AdminConfig(StatesGroup):
    main = State()
    invite_bonus = State()
    ref_bonus = State()
    warns = State()


class AdminSeat(StatesGroup):
    main = State()
    add = State()


class AdminSeatSetting(StatesGroup):
    main = State()
    name = State()
    max_count = State()
    photo = State()
    delete = State()


class AdminCategory(StatesGroup):
    main = State()
    add = State()
    info = State()
    change_name = State()
    change_desc = State()


class AdminProduct(StatesGroup):
    info = State()
    add_name = State()
    add_price = State()
    change_name = State()
    change_desc = State()
    change_help = State()
    change_price = State()
    change_photo = State()
    consumable = State()
    consumable_amount = State()


class AdminConsumable(StatesGroup):
    main = State()
    add_name = State()
    info = State()
    change_name = State()
    change_price = State()
    change_total = State()
    change_unit = State()


class AdminWorkSeat(StatesGroup):
    main = State()
    team_choice = State()

    class SeatActions(StatesGroup):
        main = State()
        move = State()
        join = State()
        sales_choice = State()

    class AddSale(StatesGroup):
        category_choice = State()
        product_choice = State()

    class SalesActions(StatesGroup):
        main = State()
        set_discount = State()
        pay = State()


class AdminStat(StatesGroup):
    start_date = State()
    end_date = State()
